Skip to content

feat: support setting joint names for each state dim in URDF viz#108

Open
shuheng-liu wants to merge 3 commits intomainfrom
feat/urdf-joint-order
Open

feat: support setting joint names for each state dim in URDF viz#108
shuheng-liu wants to merge 3 commits intomainfrom
feat/urdf-joint-order

Conversation

@shuheng-liu
Copy link
Collaborator

What this does

This PR supports setting joint names for each state dim in URDF visualization. If unset, will default to the names in info.json of the dataset.

How it was tested

Ran URDF visaulization script on a local dataset with and without --joint-names arg

How to checkout & try? (for the reviewer)

Run URDF visaulization script on a local dataset with and without --joint-names arg

Checklist

  • I have added Google-style docstrings to important functions and ensured function parameters are typed.
  • My PR includes policy-related changes.
    • If the above is checked: I have run the GPU pytests (pytest -m "gpu") and regression tests.

Note: Before submitting this PR, please read the contributor guideline.

@shuheng-liu shuheng-liu self-assigned this Feb 7, 2026
@shuheng-liu shuheng-liu added the feature New feature or request label Feb 7, 2026
Copilot AI review requested due to automatic review settings February 7, 2026 09:32
Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR adds support to the dataset URDF visualization script for associating each observed state dimension with a specific URDF joint name, defaulting to dataset metadata when not explicitly provided.

Changes:

  • Added a joint_names parameter to visualize_dataset() and used it to label/log state scalars and drive URDF joint transforms by name.
  • Added --joint-names CLI argument (comma-separated) and wired it into the script’s main() argument handling.
  • Switched URDF joint lookup from positional indexing to a name→joint mapping.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

shuheng-liu and others added 2 commits February 7, 2026 15:03
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

feature New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant